﻿using System;
using System.Text;
using System.Collections.Generic;
using System.Linq;
using Microsoft.VisualStudio.TestTools.UnitTesting;

namespace Cips.DataMining.Tests
{
    /// <summary>
    /// Summary description for TermVectorTests
    /// </summary>
    [TestClass]
    public class TermVectorTests
    {
        public TermVectorTests()
        {
        }

        /// <summary>
        ///Gets or sets the test context which provides
        ///information about and functionality for the current test run.
        ///</summary>
        public TestContext TestContext { get; set; }

        #region Additional test attributes
        //
        // You can use the following additional attributes as you write your tests:
        //
        // Use ClassInitialize to run code before running the first test in the class
        // [ClassInitialize()]
        // public static void MyClassInitialize(TestContext testContext) { }
        //
        // Use ClassCleanup to run code after all tests in a class have run
        // [ClassCleanup()]
        // public static void MyClassCleanup() { }
        //
        // Use TestInitialize to run code before running each test 
        // [TestInitialize()]
        // public void MyTestInitialize() { }
        //
        // Use TestCleanup to run code after each test has run
        // [TestCleanup()]
        // public void MyTestCleanup() { }
        //
        #endregion

        [TestMethod]
        public void BasicTests()
        {
            var dictionary = new TermDictionary<string>();
            var vector = new SparseTermVector<string>(dictionary);

        	var dummy = dictionary["filler"];

			Assert.AreEqual(0, vector["test"]);

			vector["test"] = 1;
			vector["second"] = 2;

			Assert.AreEqual(1, vector["test"]);
			Assert.AreEqual(2, vector["second"]);

        	vector[dictionary[1]]++;

			Assert.AreEqual("test", dictionary[1]);
			Assert.AreEqual(2, vector["test"]);

			CollectionAssert.AreEquivalent(new[] { 1, 2 }, vector.Select(i => i.Key).ToList());
        }
    }
}
